package com.heima.leetcode.practice;

/**
 * @author 勾新杰
 * @version 1.0
 * @description: leetcode 75.颜色分类
 * @date 2024/12/21 11:17
 */
public class E75 {

    /**
     * <h3>方法一：划分区域，时间复杂度：O(N),空间复杂度：O(1)</h3>
     *
     * @param nums 数组
     */
    public void sortColors1(int[] nums) {
        // 1. 两个变量：low 0区域的右边界（不包含）, high 2区域的左边界（不包含）
        int low = 0, high = nums.length - 1;
        for (int i = 0; i <= high; i++) { // 当等于high的时候也要检查
            switch (nums[i]) {
                case 0 -> {
                    if (i != low) swap(nums, low, i);
                    low++;
                    // 这里i不用--，因为如果前面如果右i=2的已经换到后面去了
                }
                case 1 -> {
                }
                case 2 -> {
                    if (i != high) swap(nums, high, i);
                    high--;
                    i--;
                }
            }
        }
    }

    /**
     * 交换两个元素
     *
     * @param nums   数组
     * @param index1 索引1
     * @param index2 索引2
     */
    private void swap(int[] nums, int index1, int index2) {
        int temp = nums[index1];
        nums[index1] = nums[index2];
        nums[index2] = temp;
    }

    /**
     * <h3>方法二：计数，时间复杂度：O(N)，空间复杂度：O(1)</h3>
     *
     * @param nums 数组
     */
    public void sortColors2(int[] nums) {
        int[] count = new int[3];
        for (int num : nums) count[num]++;
        int index = 0;
        for (int i = 0; i < count.length; i++)
            while (count[i]-- > 0) nums[index++] = i;
    }
}
